Add comments and docs