Migrate to WPF