Add vertex attribute setting code