Update db_to_data.py

+ added the functionality to load particle ID and charge density from the db