Bug fix
Added script for figure of PYAS performance
Updated notebooks