Make slight changes to controllers