Small changes to the code