small modifications