Track weight of rep

Currently crashes if no weight is input - this needs to be made more
robust