Improve data module