made some changes and added red square and pentagon