Added new test & edited test loop