Did some styling to clean things up a bit more