Added GUI and better function, for real