Got rid of unnecessary "return"