diff --git a/mispr/lammps/database.py b/mispr/lammps/database.py
index 7ae3bc04..f0099361 100644
--- a/mispr/lammps/database.py
+++ b/mispr/lammps/database.py
@@ -1,6 +1,4 @@
-# coding: utf-8
-
-# Defines classes for handling mongodb interactions.
+"""Define classes for handling mongodb interactions."""
 
 import sys
 import logging
diff --git a/mispr/lammps/firetasks/parse_outputs.py b/mispr/lammps/firetasks/parse_outputs.py
index af84496e..7e968496 100644
--- a/mispr/lammps/firetasks/parse_outputs.py
+++ b/mispr/lammps/firetasks/parse_outputs.py
@@ -1,7 +1,4 @@
-# coding: utf-8
-
-
-# Defines firetasks for reading output of Ambertools programs and LAMMPS.
+"""Define firetasks for reading output of Ambertools programs and LAMMPS."""
 
 import os
 import json
diff --git a/mispr/lammps/firetasks/run.py b/mispr/lammps/firetasks/run.py
index 3d537100..330129f6 100644
--- a/mispr/lammps/firetasks/run.py
+++ b/mispr/lammps/firetasks/run.py
@@ -1,7 +1,4 @@
-# coding: utf-8
-
-
-# Defines firetasks for running LAMMPS simulations and AmberTools.
+"""Define firetasks for running LAMMPS simulations and AmberTools."""
 
 import os
 import re
diff --git a/mispr/lammps/firetasks/write_inputs.py b/mispr/lammps/firetasks/write_inputs.py
index 17d6e325..0302334b 100644
--- a/mispr/lammps/firetasks/write_inputs.py
+++ b/mispr/lammps/firetasks/write_inputs.py
@@ -1,7 +1,4 @@
-# coding: utf-8
-
-
-# Defines firetasks for writing LAMMPS input files.
+"""Define firetasks for writing LAMMPS input files."""
 
 import os
 import json
diff --git a/mispr/lammps/fireworks/core.py b/mispr/lammps/fireworks/core.py
index dd8ab59a..f58be0f3 100644
--- a/mispr/lammps/fireworks/core.py
+++ b/mispr/lammps/fireworks/core.py
@@ -1,7 +1,4 @@
-# coding: utf-8
-
-
-# Defines core fireworks for ambertools and lammps.
+"""Define core fireworks for ambertools and lammps."""
 
 import os
 import logging
diff --git a/mispr/lammps/utilities/opls.py b/mispr/lammps/utilities/opls.py
index 2290af23..8d13bb84 100644
--- a/mispr/lammps/utilities/opls.py
+++ b/mispr/lammps/utilities/opls.py
@@ -1,8 +1,5 @@
-# coding: utf-8
-
-
-# Implements a core class MaestroRunner for assigning force field
-# parameters on a molecule using Maestro software.
+"""Implements a core class MaestroRunner for assigning force field parameters on a
+molecule using Maestro software."""
 
 import os
 import re
diff --git a/mispr/lammps/utilities/utilities.py b/mispr/lammps/utilities/utilities.py
index a6422182..cd106c7a 100644
--- a/mispr/lammps/utilities/utilities.py
+++ b/mispr/lammps/utilities/utilities.py
@@ -1,7 +1,4 @@
-# coding: utf-8
-
-
-# Defines lammps utility functions.
+"""Define lammps utility functions."""
 
 import os
 import json
diff --git a/mispr/lammps/workflows/base.py b/mispr/lammps/workflows/base.py
index 64015142..de2d5eef 100644
--- a/mispr/lammps/workflows/base.py
+++ b/mispr/lammps/workflows/base.py
@@ -1,7 +1,4 @@
-# coding: utf-8
-
-
-# Defines standard lammps workflows.
+"""Define standard lammps workflows."""
 
 import os
 
diff --git a/setup.py b/setup.py
index eb80adfd..3ca28651 100644
--- a/setup.py
+++ b/setup.py
@@ -14,17 +14,20 @@
     url="https://github.com/molmd/mispr",
     install_requires=[
         "numpy >= 1.21.1",
+        # "pymongo >= 3.11.0",
+        # "pymongo <= 3.12.0",
         "pymongo>=3.3.0,<=3.12.0",
         "matplotlib >= 3.3.1",
         "networkx >= 2.5",
-        "fireworks >= 1.9.6",
-        "monty >= 4.0.0",
+        "FireWorks >= 2.0.3",
+        # "monty >= 4.0.0",
         "scipy >= 1.5.2",
         "pandas >= 1.1.2",
         "pubchempy",
         "parmed",
         "mdproptools",
-        "dnspython"
+        "dnspython",
+        "ruamel.yaml>=0.15.35,<=0.17.40",  # should be removed in the future, but is needed since FireWorks has not updated their function calls to match newer versions of ruamel.yaml
     ],
     packages=setuptools.find_packages(),
     classifiers=[