Adding tests for homework3