Change to class based view