Fixed styling and added functionality.
Merge pull request #15 from b91961/master

Fixed styling and added functionality.