Update to current flixel API