update to clean up stuff