Add task04 modules to fix
Fix code style