Added class stub for migration