Small style updates, add search