Try adding Serializable trait