Modify code to Ruby Style Guide rules