Very simple Python gravity simulation