Working on prob version to include planes