Got my code mixed up