Finishing the code test!