Context generalization in Drosophila visual learning requires the mushroom bodies