Fixed #111, minor styling fix