Need to fix my code.