Style and update documentation