Update styling
Merge pull request #4 from middric/styles

Update styling and add comments