Edited code style