Code works, need to be fine adjusted