cleaned up comments
added extra features
Added working GUI