Added completed code for lab 4