Correction in the code, now it fits perfectly