Updated with add method