Design of shaft subjected to dynamic load