Corrected mistake where i put code in wrong file