Prime: mini refactor