Changed how program takes input