improved UI + todos