Refactored update action and added comments box