UI brushup.
Merge pull request #186 from jasvir99/dirty

UI brushup.