Removed unnecessary return
Added missing return tag