Small changes to last commit