Minor feautures add