Add weight.py
Update comments in each file