Clean up comments.
Update types.

Update formatting and types.
Update type signature.

Update type signature to be correct.