Added a remove weights function