Fixed code and added section