Trimmed some of the if/else statement. Still works