code working

finished. code working!