Adding proper migration script.