Fixed code styling after review