Update classes used to style info, warning messages

Additional fixes #1