6. Creating a View Class (bug-fix)