Handle input better.