School assigment - battleship game