started to add mass euqation