Added Message model and edited controller for testing