Add basic collision logic