Still working on gui