Wrote a remove from list method, will improve