Improved GUI, fixed some bugs.