Fix TS types

Previously setup and createMedia were (mis)typed as returning another function. This has been corrected and the types file has been cleaned up to (hopefully) be a bit more understandable.