added personal code to assignment