updated the gui and added the code to branch yue