updated code

dev work