Cleanup code.
Simple update to run command.