made docs more clear
added proper returns