Working on tests
Assignment 06 completed