Working with different file formats in python