Cleanup startup class by using partial classes.
Added sources to make restore work.