ADDED: Migration logic to update model from 0.1 to 0.2