added description for remove
cleaned code