Added code to take input