code modifications for dropout rate plot