Added up to functionality in notebook