Added CSS classes matching original ones

Thanks