Minor cleanup
Merge branch 'cleanup' into staging