Refactored for easier to show different messages