small modifications to  previous step