Merge pull request #1 from sidharthsurana/fix-ubuntu

Fix the issue related to different python versions