small code structure refactor in colony system