GUI working, but not soooo nice...working on it