Imported register module, got rid of imported math