Render customClass if it defined

Hi Fede, sorry for broke your code, here is the fix. Please check it and refactor as you want.