minor updates
minor code updates