Done code as per comment