Polish code and add README also