Add basic collision code