Updated message function and created unit tests for it