updated code, things are working