Condense the code a bit