Como calcular a Energia Potencial Gravitacional de um corpo dado a sua massa e altura em Python

# função principal do programa
def main():
  # gravidade terrestre em m/s2
  gravidade = 9.80665
  # massa do corpo
  massa = 2 # em kg
  # altura do corpo em relação ao chão
  altura = 1.5 # em metros
  
  # vamos calcular a energia potencial gravitacional
  epg = massa * gravidade * altura
  
  # mostramos o resultado
  print("A Energia Potencial Gravitacional é: {0}J".format(epg))
  
if __name__== "__main__":
  main()