tidy some code - no sig changes