Refactor email prediction spec