Added loading state to comment button