improved ui
Merge pull request #4 from NathanielSchmitz/master

improved ui