Small changes to the code, no output