Design tweaks and add gravity for now