added my code to my fork