improves GUI, now it's working