Design of spatial variability in thermal energy storage modules for enhanced power density