small modifications

added small changes