Minor - improving code.