Adding tests for homework 3