Fixed migration and created milestone class