cleaned up code to remove code not needed for step 2