cleanup
small update to simplify