Added // MARK: - directive in order to achieve a better code organization