Corrected element() to adjust for change in classes()
Added Unit Test for elements()