Added module use and fixed code formatting for shell.