Updated with a loop based on user num_layers input and added prompt for num padding
Add padding feature
First functional version