Added tests for week 4