Final stage- GUI still needs a little work