Fixed Code (My bad man)