Den modifierade main koden som sparar pot., kin- och totala energin