Slight nicety refactor.  Trying to figure out a way to make this extensible instead of hard coded...would be a mess to do but I can give it a shot.