Simplifiqué el código pero aún no conserva la energía