Added code to main class, Ocean and OceanTest.