Replace the background with a class