README improvements and UI bug fix