Fixed code on second exercise