fixed assignment 2