diff --git a/src/HeadlessWrapper.lua b/src/HeadlessWrapper.lua
index 12e61024b..683afd95c 100644
--- a/src/HeadlessWrapper.lua
+++ b/src/HeadlessWrapper.lua
@@ -44,7 +44,7 @@ function imageHandleClass:ImageSize()
 end
 
 -- Rendering
-function RenderInit() end
+function RenderInit(flag, ...) end
 function GetScreenSize()
 	return 1920, 1080
 end
@@ -157,6 +157,7 @@ function OpenURL(url) end
 function SetProfiling(isEnabled) end
 function Restart() end
 function Exit() end
+function TakeScreenshot() end
 
 ---@return string? provider
 ---@return string? version
diff --git a/src/Launch.lua b/src/Launch.lua
index 0b579a4c3..dffeb8307 100644
--- a/src/Launch.lua
+++ b/src/Launch.lua
@@ -250,7 +250,7 @@ end
 ---Download the given page in the background, and calls the provided callback function when done:
 ---@param url string
 ---@param callback fun(response:table, errMsg:string) @ response = { header, body }
----@param params table @ params = { header, body }
+---@param params? table @ params = { header, body }
 function launch:DownloadPage(url, callback, params)
 	params = params or {}
 	local script = [[
diff --git a/src/Modules/BuildSiteTools.lua b/src/Modules/BuildSiteTools.lua
index 67f1d4b0a..9bd1d7575 100644
--- a/src/Modules/BuildSiteTools.lua
+++ b/src/Modules/BuildSiteTools.lua
@@ -32,8 +32,8 @@ buildSites.websiteList = {
 }
 
 --- Uploads a PoB build code to a website
---- @param websiteInfo Table Contains the postUrl, any postParams, and a prefix to add to the response
---- @param buildCode String The build code that will be uploaded
+--- @param websiteInfo table Contains the postUrl, any postParams, and a prefix to add to the response
+--- @param buildCode string The build code that will be uploaded
 function buildSites.UploadBuild(buildCode, websiteInfo)
 	local response
 	if websiteInfo then
@@ -72,9 +72,9 @@ function buildSites.UploadBuild(buildCode, websiteInfo)
 end
 
 --- Downloads a PoB build code from a website
---- @param link String A link to the site that contains the link to the raw build code
---- @param websiteInfo Table Contains the downloadUrl
---- @param callback Function The function to call when the download is complete
+--- @param link string A link to the site that contains the link to the raw build code
+--- @param websiteInfo table? Contains the downloadUrl
+--- @param callback function The function to call when the download is complete
 function buildSites.DownloadBuild(link, websiteInfo, callback)
 	local siteCodeURL
 	-- Only called on program start via protocol handler
@@ -103,7 +103,7 @@ function buildSites.DownloadBuild(link, websiteInfo, callback)
 end
 
 -- Parses and converts URI's to import links. Currently only supports protocol handler URI's, extend as needed.
--- @param uri String Example: pob2://pobbin/<id> or pob2://poeninja/<id>
+--- @param uri string Example: pob2://pobbin/<id> or pob2://poeninja/<id>
 function buildSites.ParseImportLinkFromURI(uri)
 	local importLink = nil