TODO fix input class