Cleaning up python wrapper library.