UI fix
Merge pull request #82 from benitoro/master

UI fix