Added skeleton code for step 2