Added collision code, sortof works