formatting code better