move update code to specific class