Gravitational force added but still need to polished