Refactord to make comments polymorphic
-Tested in rake console, everything seems good!