Added another migration case
Added migration wrapper class