Code cleanup

Minor code cleanup