Tweaked if statement to handle options better.