made a better ui fixes #2 fixes #3