Changed how the grab function works