Change Activity to Class

Refer to the Class instead of the Activity