Small enhancements

style and ui changes

Signed-off-by: Joshua Pekera <joshuapekera@me.com>