Added classes for assignment 5