Adds classes for program

Adds main class based on homework assignment criteria.