Adding tests for homework 5