If statements work, but still need work