Added custom init params functionality
Updated