added solution to first test