adding what's needed to pass test