More changes to UI to make it work a little better (still not perfect)
  Need to work on the Markov model later.