Add extra class to make styling easier