GUI has been slitly improved