Updated gravity input and added regression test