Add comment and little refactory