Add the code in the main activity