Removed unnecessary return
Made more accurate