complete implement the classes assignment