updated the headers and added cpp files for unit and cube