small improvenents