Fixed comments and small changes
More PEP8 compliant, fixed small bug and small changes in code
Added custom_types for tag groups, TODO: make unit tests for that