Smart Interview Basic Hackerrank Solutions Python