Wokring on gravity code