small project refactor