Refactor to create BlueGreen class and added some tests. Closes #4