Small changes to source