Refactor 'Init' class