Creating a view class
Synced with correct code from section 6