corrections/ adding second part of the test