Updated with new code

Issue with versions