2 files changed, 14 insertions, 13 deletions
diff --git a/vfd-mavlink/src/main/scala/org/mavlink/messages/Message.scala b/vfd-mavlink/src/main/scala/org/mavlink/messages/Message.scala
index 91e446c..f22d0b3 100644
--- a/vfd-mavlink/src/main/scala/org/mavlink/messages/Message.scala
+++ b/vfd-mavlink/src/main/scala/org/mavlink/messages/Message.scala
@@ -1,16 +1,8 @@
 package org.mavlink.messages
 
-trait Message {}
+import org.mavlink.PayloadReader
+import org.mavlink.Packet
+import org.mavlink.PayloadBuilder
 
-case class Heartbeat(
-  `type`: Int,
-  autopilot: Int,
-  baseMode: Int,
-  customMode: Int,
-  systemStatus: Int,
-  mavlinkVersion: Int
-) extends Message
-
-object Message {
-
-}
-\ No newline at end of file
+trait Message
+  
+\ No newline at end of file
diff --git a/vfd-mavlink/src/main/scala/org/mavlink/messages/messages.scala b/vfd-mavlink/src/main/scala/org/mavlink/messages/messages.scala
new file mode 100644
index 0000000..422949f
--- /dev/null
+++ b/vfd-mavlink/src/main/scala/org/mavlink/messages/messages.scala
@@ -0,0 +1,9 @@
+package org.mavlink.messages
+
+case class Heartbeat(
+  customMode: Int,
+  `type`: Byte,
+  autopilot: Byte,
+  baseMode: Byte,
+  systemStatus: Byte,
+  mavlinkVersion: Byte) extends Message
+\ No newline at end of file