Use of split to take input in single line and use of f for formatting of string in python 3.6