completed step 11: fixed bug