Condensed the code a bit