Code cleanup/fixes, enum updates