Refactor message classes

**Description** Move message classes to their own package and organize them a bit. Also, organize the imports.
**Testing** `sbt test`