Tests for task 5 added