Updates to battleship class and b_main function