how to check netowrk either mobile network or wifi

public class MainActivity extends AppCompatActivity {
    Button btnclick;
    TextView textView;
    Button wifiEnabled;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        btnclick = (Button) findViewById(R.id.btnClick);
        textView = (TextView) findViewById(R.id.textViews);
        wifiEnabled=(Button)findViewById(R.id.btnwifienbled);
        ConnectivityManager connect = (ConnectivityManager) getApplicationContext().getSystemService(Context.CONNECTIVITY_SERVICE);
        NetworkInfo networke = connect.getActiveNetworkInfo();
        if (networke != null) {
            if (networke.getType() == ConnectivityManager.TYPE_MOBILE)
            {
                Toast.makeText(getApplicationContext(), "i am mobile network", Toast.LENGTH_SHORT).show();

            }
            else if (networke.getType() == ConnectivityManager.TYPE_WIFI)
            {
                Toast.makeText(getApplicationContext(), "i am wifi nework", Toast.LENGTH_SHORT).show();
            }

            else
                {
                Toast.makeText(getApplicationContext(), "network not connected", Toast.LENGTH_SHORT).show();
            }
        }
            btnclick.setOnClickListener(new View.OnClickListener() {
                @Override
                public void onClick(View v) {
                    Intent in=getIntent();
                    finish();
                    startActivity(in);
                }
            });
        wifiEnabled.setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View v) {
                startActivity(new Intent(Settings.ACTION_SETTINGS));
            }
        });
    }
}