add some code in .m file