Adjusted code to future division