Clean up weights