Two codes not working after new layout