Added controller and made simple working demo