Modified plain to class based view