Adding tests for the 4th homework