update some code for this pro