add style and better ux
Merge pull request #41 from vinitkumar/fix/styling

add style and better ux