fix assignment4
added solution for assignment 5