Added imagine bundle and fixed test