Infosys Coding Questions Python