Add initial migration and some tests