add logic into helper class