Added a bit more to the gui