added test data for new function
added basic if-statement