works. need to fix gui