Changed where code should go. THINKING BOUT IT