Just a small change in initialization of GUI