added return statements for tests