style improvments

style improvments