adding return to both functions