Adding the required changes