Fixed bug in step 2