Right idea, wrong if statement...