New functionality added to the code