Adding test lab, writing codingbat function