Added function args and return