Tests for homework4 added