Different colours for warning and success messages

Closes #5