Design and modeling of shaft and impeller