Improves a bit the gui