Fixed new css to account for custom class

@amitaibu pls look into the custom class code