quick little refactor