adding basic python code for handling data