Some UI fixes.
UI changes.
Merge pull request #27 from emtwo/unit-tests

Unit tests