my solution for assignment 2