Switch to class based views