Change the code into my style