Get input from STDIN instead of being hardcoded