Simplified battle ship game in python