finalize new UI! Fixes #11