A few improvements and modification for compatibility with Timepix3 data