Organized code a little better by adding an "else if"