uodating and testing out code to get it to work
added argument and ran tests with code untill I could make it work