Diana's code without gravity