Update tests to work with mastermind.rb
Refactor with mentor Geoff