Separate tests from homework implementation