correcting functionality added