Update setup.py

added writemoment module