Edited return statement