cleaned code added prettiness