modify main and input class