Updated version for the added feature