Search unit function done, just need some design fix. :)