Changed the .h and .m file