Some modifications to the code