Update

Minor bug fixes.