code cleanup. 

Add one more function.