edits to red green refactor