- working version! -- just change gui haha