Remove debug code, add input