Random walk using python