Finished and working

Took longer than is should have to work out that I need to use
.equals() rather than == now that I have switched to java.