Removed todo comment for adding custom capability