Changed the KeyMap method parameter types to 'int'
Added DI for Loop and Stage constructors