Added some code in the main activity