Refactoring Changes

made changes to the code in order to make the code more scalable. The
code now allows for more than just red and blue colors to be used. The
machine learning method still needs to be updated.