Added missing import for Bundle class