python gravity simulation