Fixed indentation
Added floor with a loop