Now is the code correct, only missing a ship