Added lines to test