Bugfix

Changed the method to incorporate the required integer, added an unused import and fixed the spacing.