WIP: classification script