could use "elif" here, and trim off a line.  Mostly a style thing