small changes to walk function. WORKING