Code working but remove method not implemented yet