2d List Python Based On Input