small code updates