Added callbacks for dragging + better look