add tests for 4th task