Cleanup

Minor update