Updated GUI to look like previous version