Updated Code

Updated code to match the slab accurately