Small class style change