worked on fixing Step 2