Polished UI a bit and added functionality.