Model your Write-Log entries after the other entries in the toolkit. If you have a custom function you wrote, then make sure to get the name of the function into a variable (ex: ${CmdletName}) like the functions in the toolkit do and then make sure to pass this parameter to Write-Log: -Source ${CmdletName}
Also, make sure you configured the XML config to create CMTrace format log files if you want the colors.
Also, make sure you configured the XML config to create CMTrace format log files if you want the colors.