Code added

Updated docs