Updated initial test to cover every line