Add the architecture describtion and modify the learning rate to be higher, and improve the GUI