Added classes as a visual for the screens in Milestone 2