Fixing assignment 2 instructions