GUI working (partially)